print square of numbers in python